轻量级线程

概述

在并发编程中，线程是实现并行执行任务的基本单位。传统的线程模型通常依赖于操作系统提供的线程，这些线程由操作系统内核管理，称为“重量级线程”。然而，重量级线程在创建和管理上开销较大，且容易受到操作系统调度策略的影响。为了解决这些问题，Haskell 提供了轻量级线程（Lightweight Threads），也称为“绿色线程”或“用户级线程”。

轻量级线程由 Haskell 运行时系统（GHC 运行时）管理，而不是操作系统。它们在用户空间中创建和管理，因此开销较小，且可以更灵活地调度。轻量级线程非常适合处理 I/O 密集型任务，如网络通信和文件操作。

轻量级线程的基本概念

1. 创建轻量级线程

在 Haskell 中，可以使用 forkIO 函数来创建轻量级线程。forkIO 函数接受一个 IO 动作作为参数，并在新的轻量级线程中执行该动作。

import Control.Concurrent

main :: IO ()
main = do
    putStrLn "Main thread starting..."
    forkIO $ do
        putStrLn "Child thread starting..."
        threadDelay 1000000  -- 延迟 1 秒
        putStrLn "Child thread finished."
    putStrLn "Main thread finished."

在这个例子中，forkIO 创建了一个新的轻量级线程，该线程执行一个简单的 IO 动作。主线程和子线程并发执行，主线程在子线程完成之前就已经结束。

2. 线程间通信

轻量级线程之间可以通过 MVar 进行通信。MVar 是一个同步原语，类似于一个可以为空的容器。线程可以从 MVar 中读取数据或向其中写入数据。

import Control.Concurrent

main :: IO ()
main = do
    mvar <- newEmptyMVar
    forkIO $ do
        putStrLn "Child thread starting..."
        threadDelay 1000000  -- 延迟 1 秒
        putMVar mvar "Hello from child thread!"
    putStrLn "Main thread waiting for message..."
    message <- takeMVar mvar
    putStrLn ("Received message: " ++ message)

在这个例子中，主线程创建了一个空的 MVar，并启动了一个子线程。子线程在延迟 1 秒后向 MVar 中写入一条消息。主线程在等待 MVar 中的消息，一旦收到消息，就将其打印出来。

3. 线程同步

MVar 不仅可以用于线程间通信，还可以用于线程同步。例如，可以使用 MVar 来确保多个线程按顺序执行某些操作。

import Control.Concurrent

main :: IO ()
main = do
    mvar <- newEmptyMVar
    forkIO $ do
        putStrLn "Child thread 1 starting..."
        threadDelay 1000000  -- 延迟 1 秒
        putStrLn "Child thread 1 finished."
        putMVar mvar ()
    forkIO $ do
        putStrLn "Child thread 2 waiting..."
        takeMVar mvar
        putStrLn "Child thread 2 starting..."
        threadDelay 1000000  -- 延迟 1 秒
        putStrLn "Child thread 2 finished."
    putStrLn "Main thread finished."

在这个例子中，主线程创建了一个空的 MVar，并启动了两个子线程。第一个子线程在完成任务后向 MVar 中写入一个空值，第二个子线程在等待 MVar 中的值，确保在第一个子线程完成后才开始执行。

实践练习

练习 1: 并发计算

编写一个 Haskell 程序，创建多个轻量级线程来计算斐波那契数列。每个线程计算一个不同的斐波那契数，并将结果存储在一个列表中。主线程等待所有子线程完成后，打印出所有计算结果。

练习 2: 生产者-消费者模型

实现一个简单的生产者-消费者模型。生产者线程生成随机数，并将它们放入一个 MVar 中。消费者线程从 MVar 中取出这些数，并计算它们的平均值。主线程等待所有消费者线程完成后，打印出平均值。

总结

轻量级线程是 Haskell 并发编程中的一个强大工具，它们提供了高效、灵活的并发执行方式。通过 forkIO 创建线程，使用 MVar 进行线程间通信和同步，可以实现复杂的并发任务。掌握轻量级线程的使用，将帮助你更好地利用 Haskell 的并发能力，构建高性能的应用程序。